package com.prawn.user.service;

import java.util.*;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Expression;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import javax.persistence.criteria.Selection;

import com.prawn.user.dao.MenuDao;
import com.prawn.user.dao.RolePermissionDao;
import com.prawn.user.pojo.Menu;
import com.prawn.user.pojo.RolePermission;
import com.prawn.user.pojo.RolePermissionPK;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;
import util.IdWorker;

import com.prawn.user.dao.RoleDao;
import com.prawn.user.pojo.Role;

/**
 * 服务层
 * 
 * @author Administrator
 *
 */
@Service
public class RoleService {

	@Autowired
	private RoleDao roleDao;

	@Autowired
	private RolePermissionDao rolePermissionDao;

	@Autowired
	private MenuDao menuDao;
	
	@Autowired
	private IdWorker idWorker;

	/**
	 * 查询全部列表
	 * @return
	 */
	public List<Role> findAll() {
		return roleDao.findAll();
	}

	
	/**
	 * 条件查询+分页
	 * @param whereMap
	 * @param page
	 * @param size
	 * @return
	 */
	public Page<Role> findSearch(Map whereMap, int page, int size) {
		Specification<Role> specification = createSpecification(whereMap);
		PageRequest pageRequest =  PageRequest.of(page-1, size);
		return roleDao.findAll(specification, pageRequest);
	}

	
	/**
	 * 条件查询
	 * @param whereMap
	 * @return
	 */
	public List<Role> findSearch(Map whereMap) {
		Specification<Role> specification = createSpecification(whereMap);
		return roleDao.findAll(specification);
	}


	/**
	 * 查询角色信息和对应的权限并生成菜单树
	 * @param roleId
	 * @return
	 */
	public List<Menu> findPermissionByRoleId(String roleId) {
		//查询所有角色权限信息
		String[] permissions = findById(roleId).getPermissions();
		List<Menu> menuList = menuDao.findByIdInOrderBySortDesc(permissions);

		//查询其中的顶级菜单
		List<Menu> rootMenuList = new ArrayList<>();
		for (Menu menu: menuList) {
			if (menu.getParentId().equals("0")) {
				rootMenuList.add(menu);
			}
		}

		//递归查询菜单子列表
		for (Menu menu : rootMenuList) {
			menu.setChildren(getChild(menu.getId(),menuList));
		}

		return rootMenuList;
	}

	/**
	 * 查询角色信息和对应的权限
	 * @param id
	 * @return
	 */
	public Role findById(String id) {
		//查询角色信息
		Role role = roleDao.findById(id).get();
		//查询角色的权限信息列表
		List<RolePermission> rolePermissions = rolePermissionDao.findByRoleIdEquals(id);
		String[] permissions = new String[rolePermissions.size()];
		for (int i = 0;i < rolePermissions.size();i++) {
			permissions[i] = rolePermissions.get(i).getMenuId();
		}
		role.setPermissions(permissions);
		return role;
//		return roleDao.findById(id).get();
	}

	/**
	 * 增加
	 * @param role
	 */
	@Transactional
	public void add(Role role) {
		role.setId( idWorker.nextId()+"" );
		//添加权限信息
		String[] menuIds = role.getPermissions();
		RolePermission roleMenuKey = new RolePermission();
		String roleId = role.getId();
		for (String menuId : menuIds) {
			roleMenuKey.setRoleId(roleId);
			roleMenuKey.setMenuId(menuId);
			rolePermissionDao.save(roleMenuKey);
		}
		roleDao.save(role);
	}

	/**
	 * 修改
	 * @param role
	 */
	@Transactional
	public void update(Role role) {
		//删除角色原有的权限信息再添加
		String[] permissions = role.getPermissions();
		String roleId = role.getId();
		rolePermissionDao.deleteByRoleId(roleId);
		RolePermission roleMenuKey = new RolePermission();
		for (String menuId : permissions) {
			roleMenuKey.setRoleId(roleId);
			roleMenuKey.setMenuId(menuId);
			rolePermissionDao.save(roleMenuKey);
		}
		//修改角色信息
		roleDao.save(role);
	}

	/**
	 * 删除
	 * @param id
	 */
	@Transactional
	public void deleteById(String id) {
		//删除权限信息
		rolePermissionDao.deleteByRoleId(id);
		roleDao.deleteById(id);
	}

	/**
	 * 动态条件构建
	 * @param searchMap
	 * @return
	 */
	private Specification<Role> createSpecification(Map searchMap) {

		return new Specification<Role>() {

			@Override
			public Predicate toPredicate(Root<Role> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
				List<Predicate> predicateList = new ArrayList<Predicate>();
                // 归属机构
                if (searchMap.get("officeId")!=null && !"".equals(searchMap.get("officeId"))) {
                	predicateList.add(cb.equal(root.get("officeId").as(String.class), (String)searchMap.get("officeId")));
                }
                // 角色名称
                if (searchMap.get("name")!=null && !"".equals(searchMap.get("name"))) {
                	predicateList.add(cb.equal(root.get("name").as(String.class), (String)searchMap.get("name")));
                }
                // 英文名称
                if (searchMap.get("enname")!=null && !"".equals(searchMap.get("enname"))) {
                	predicateList.add(cb.like(root.get("enname").as(String.class), "%"+(String)searchMap.get("enname")+"%"));
                }
                // 角色类型
                if (searchMap.get("roleType")!=null && !"".equals(searchMap.get("roleType"))) {
                	predicateList.add(cb.equal(root.get("roleType").as(String.class), (String)searchMap.get("roleType")));
                }
                // 是否可用（0可用）
                if (searchMap.get("useable")!=null && !"".equals(searchMap.get("useable"))) {
                	predicateList.add(cb.equal(root.get("useable").as(String.class), (String)searchMap.get("useable")));
                }
				return cb.and( predicateList.toArray(new Predicate[predicateList.size()]));

			}
		};

	}


	/**
	 * 递归查询子菜单列表
	 * @param id
	 * @param menuList
	 * @return
	 */
	private List<Menu> getChild(String id,List<Menu> menuList) {
		// 子菜单
		List<Menu> childList = new ArrayList<>();

		// 遍历所有节点，将父菜单id与传过来的id比较
		for (Menu sysMenu : menuList) {
			if (sysMenu.getParentId() != null) {   //所有非以及菜单的都要加入chidList中（递归）
				if (sysMenu.getParentId().equals(id)) {
					childList.add(sysMenu);
				}
			}
		}

		// 把子菜单的子菜单再循环一遍
		for (Menu sysMenu : childList) {// 没有url子菜单还有子菜单（暂不考虑url的问题）
			// 递归
			sysMenu.setChildren(getChild(sysMenu.getId(), menuList));
		} // 递归退出条件
		if (childList.size() == 0) {
			return null;
		}
		return childList;
	}

}
